/*

* ------------------------------------------------------------------------
* ------------------------------------------------------------------------
* |     Smart-Shop开源商城系统/ Java商城-首选电商平台系统 请务必保留此注释头信息
* |     开源地址: https://gitee.com/ningbo-qishan/gruul-mall
* ------------------------------------------------------------------------
* ------------------------------------------------------------------------
* |     可正常分享和学习源码,未经授权 不可商用！
* |     商业版中台购买/开源版商业授权 联系技术客服
* |     官网:  https://www.73app.cn/
* ------------------------------------------------------------------------
* ------------------------------------------------------------------------
* |     Copyright (c) 2016-2999 宁波启山智软科技有限公司
* |     版权所有 ,侵权必究！
* ------------------------------------------------------------------------

*/
package com.medusa.gruul.gateway.handler;

import cn.hutool.core.util.StrUtil;
import com.medusa.gruul.common.core.auth.model.constant.Headers;
import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.springframework.stereotype.Component;

@Component
public class FeignRequestHandler implements RequestInterceptor {
    public static final ThreadLocal<String> TOKEN = new ThreadLocal<>();
    public static final ThreadLocal<String> VERSION = new ThreadLocal<>();
    public static final ThreadLocal<String> PLATFORM = new ThreadLocal<>();

    public static void set(String token,String version,String platform){
        TOKEN.set(token);
        VERSION.set(version);
        PLATFORM.set(platform);
    }

    public static void clear(){
        TOKEN.remove();
        VERSION.remove();
        PLATFORM.remove();
    }


    @Override
    public void apply(RequestTemplate requestTemplate) {
        if (StrUtil.isNotBlank(TOKEN.get())){
            requestTemplate.header(Headers.TOKEN,TOKEN.get());
        }
        if (StrUtil.isNotBlank(VERSION.get())){
            requestTemplate.header(Headers.VERSION,VERSION.get());
        }
        if (StrUtil.isNotBlank(PLATFORM.get())){
            requestTemplate.header(Headers.PLATFORM, PLATFORM.get());
        }
    }
}
